package net.cyue.ort.llm.template;

public class ModelChatMessage {
    public enum Role {
        USER,
        ASSISTANT,
        SYSTEM
    }

    public final Role role;
    public final String content;

    public ModelChatMessage(Role role, String content) {
        this.role = role;
        this.content = content;
    }

    @Override
    public String toString() {
        return this.role + ": " + this.content + "\n";
    }
}
